package com.duyus.ai.repository;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.KryoException;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;
import jakarta.annotation.PostConstruct;
import jakarta.annotation.PreDestroy;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.objenesis.strategy.StdInstantiatorStrategy;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@Component
@RequiredArgsConstructor
//@Primary
public class FileChatHistoryRepository{
    String fileDir = System.getProperty("user.dir") + "/tmp/chat-list";

    private final String BASE_DIR = fileDir;

    // 序列化
    private static final Kryo kryo = new Kryo();

    // 存储不同类型的聊天记录 ID 列表
    public Map<String, List<String>> chatHistory = new HashMap<>();

    static {
        kryo.setRegistrationRequired(false);
        // 设置实例化策略
        kryo.setInstantiatorStrategy(new StdInstantiatorStrategy());
    }

    public void save(String type, String conversationId) {
        /**
         * 如果当前类型没有对应的聊天列表，则创建一个新的空列表。
         * 如果已经存在该 chatId，不重复添加。
         * 否则加入该 chatId。
         */
        List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new ArrayList<>());
        if (chatIds.contains(conversationId)) {
            return;
        }
        chatIds.add(conversationId);
    }

    public List<String> getChatIds(String type) {
        List<String> chatIds = chatHistory.get(type);
        return chatIds == null ? List.of() : chatIds;
    }

    /**
     * 根据类型返回该类型下所有的 chatId 列表。
     * 为空的话返回空列表
     *
     * @return
     */
    @PostConstruct
    public void readChatIds() {
        File file = getConversationFile("chatHistory");
        if (file.exists()) {
            try (Input input = new Input(new FileInputStream(file))) {
                chatHistory = kryo.readObject(input, HashMap.class);
                log.info("读取文件 {}",chatHistory);
            } catch (IOException | KryoException e) {
                e.printStackTrace();
                log.info("读取文件错误");
            }
        } else {
            // 文件不存在，也初始化为空 map
            chatHistory = new HashMap<>();
            log.info("map文件不存在");
        }
    }

    @PreDestroy
    public void persistentConversationID() {
        File file = getConversationFile("chatHistory");
        try (Output output = new Output(new FileOutputStream(file))) {
            kryo.writeObject(output, chatHistory);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private File getConversationFile(String conversationId) {
        return new File(BASE_DIR, conversationId + ".kryo");
    }

}
